# 定义一个全局变量：money
money = 5000000
name = None

# 要求客户输入姓名
name = input("请输入您的姓名：")


# 定义查询函数
def query(head_shower):
    if head_shower:
        print("----------查询余额----------")
    print(f"{name}，您好，您的余额为：{money}元")


# 定义存款函数
def save():
    print("----------存款----------")
    num = int(input("请输入您的存款金额："))
    print(f"{name}，您好，您存款{num}元成功")
    global money
    money = money + num
    query(False)


# 定义取款函数
def withdrawal():
    print("----------取款----------")
    num = int(input("请输入您的取款金额:"))
    print(f"{name}，您好，您取款{num}元成功")
    global money
    money = money - num
    query(False)


# 定义主菜单栏
def main():
    print("----------主菜单----------")
    print(f"{name},您好，欢迎来到峡谷银行ATM。请选择操作：")
    print("查询余额\t[请输入1]")
    print("存款\t\t[请输入2]")
    print("取款\t\t[请输入3]")
    print("退出\t\t[请输入4]")
    return int(input("请输入您的选择："))


# 设置无限循环，确保程序不退出
while True:
    keybord_input = main()
    if keybord_input == 1:
        query(True)
        continue

    if keybord_input == 2:
        save()
        continue

    if keybord_input == 3:
        withdrawal()
        continue

    if keybord_input == 4:
        print("程序已退出")
        break
